package com.wsoft.mq.test.config.headers;

import com.wsoft.mq.test.constant.RabbitHeadersConstant;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.HeadersExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;

/**
 * headers交换机
 * @Author z001
 * @Date 2025/1/8 07:53
 */
@Configuration
public class HeadersExchangeConfig {

    /**
     * 定义队列
     * @return
     */
    @Bean
    public Queue headersQueue1() {
        return new Queue(RabbitHeadersConstant.HEADERS_QUEUE_1, true);
    }
    @Bean
    public Queue headersQueue2() {
        return new Queue(RabbitHeadersConstant.HEADERS_QUEUE_2, true);
    }

    /**
     * 定义主题交换机
     * @return
     */
    @Bean
    public HeadersExchange headersExchange() {
        // 持久化，不自动删除
        return new HeadersExchange(RabbitHeadersConstant.HEADERS_EXCHANGE, true, false);
    }


    /**
     *  创建绑定关系 1（完全匹配）
     * @return
     */
    @Bean
    public Binding headersBinding1() {
        return BindingBuilder.bind(headersQueue1())
                .to(headersExchange())
                .whereAll(headersBinding1Arguments())
                .match();
    }

    /**
     *  绑定参数 1（完全匹配）
     * @return
     */
    @Bean
    public Map<String, Object> headersBinding1Arguments() {
        Map<String, Object> arguments = new HashMap<>(3);
        arguments.put("x-match", "all");
        arguments.put("type", "report");
        arguments.put("format", "pdf");
        return arguments;
    }

    /**
     * 创建绑定关系 2（部分匹配）
     * @return
     */
    @Bean
    public Binding headersBinding2() {
        return BindingBuilder.bind(headersQueue2())
                .to(headersExchange())
                .whereAny(headersBinding2Arguments())
                .match();
    }


    /**
     * 绑定参数 2（部分匹配）
     * @return
     */
    @Bean
    public Map<String, Object> headersBinding2Arguments() {
        Map<String, Object> arguments = new HashMap<>(3);
        arguments.put("x-match", "any");
        arguments.put("type", "report");
        arguments.put("format", "pdf");
        return arguments;
    }

}
